package 二分查找;

import java.util.NavigableMap;

public class 在排序数组中查找元素的第一个和最后一个位置2 {
    //https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array/description/
    public int[] searchRange1(int[] nums, int target) {
        if(nums.length == 0) return new int[] {-1,-1};
        int left = 0, right = nums.length - 1;
        int[] ret = new int[2];
        // 查找段端点，住段的左端点
        while(left < right) { // 不能去多余搜索 left == right时的情况，否则会 中点立到住指针 进入死循环
            int mid = left + (right - left) / 2; // 中点要偏向跃段部分 即偏左
            if(nums[mid] < target) {
                left = mid + 1;
            }else {
                right = mid;
            }
        }
        if(nums[left] == target) ret[0] = left;
        else return new int[] {-1,-1};

        // 查找段端点，住段的右端点
        //left = 0; // 接下来左指针 可以直接从找到的左端点处开始 对剩下的右半部分 进行二分 查找右端点
        right = nums.length - 1;
        while(left < right) {
            int mid = left + (right - left + 1) / 2; // 中点不能偏向住段部分 即去偏右
            if(nums[mid] <= target) {
                left = mid;
            }else {
                right = mid - 1;
            }
        }
        ret[1] = left;
        return ret;
    }
    public int[] searchRange(int[] nums, int target) {
        int[] ret = new int[2];
        ret[0] = ret[1] = -1;

        if(nums.length == 0) return ret;

        // 查找段端点，住段的左端点：
        int left = 0, right = nums.length - 1;
        while(left < right) { // 不能去多余搜索 left == right时的情况，否则会 中点立到住指针 进入死循环
            int mid = left + (right - left) / 2; // 中点要偏向跃段部分 即偏左
            if(nums[mid] < target) left = mid + 1; // mid落左跃段
            else right = mid; // mid落右住段
        }

        if(nums[left] != target) return ret;
        else ret[0] = left;

        // 查找段端点，住段的右端点：
        // left = 0; // 接下来左指针 可以直接从找到的左端点处开始 对剩下的右半部分 进行二分 查找右端点
        right = nums.length - 1;
        while(left < right) {
            int mid = left + (right - left + 1) / 2; // 中点不能偏向住段部分 即偏右
            if(nums[mid] <= target) left = mid; // mid落左住段
            else right = mid - 1; // mid落右跃段
        }
        ret[1] = left;
        return ret;
    }

}
